from docx import Document
from docx.shared import Cm
from docx.enum.section import WD_ORIENTATION, WD_SECTION

def set_book_fold_margins(doc_path, output_path):
    # 打开文档
    doc = Document(doc_path)
    
    # 获取第一个节（如果没有节会自动创建一个）
    section = doc.sections[0]
    
    # 设置页面为书籍折页格式
    section.start_type = WD_SECTION.CONTINUOUS
    section.orientation = WD_ORIENTATION.LANDSCAPE  # 书籍折页实际上是横向
    
    # 设置页边距（单位为厘米）
    section.top_margin = Cm(0.5)      # 上边距 0.5cm
    section.bottom_margin = Cm(0.5)  # 下边距 0.5cm
    section.left_margin = Cm(1)      # 内侧边距 1cm（书籍折页的左页右侧/右页左侧）
    section.right_margin = Cm(0.5)    # 外侧边距 0.5cm（书籍折页的左页左侧/右页右侧）
    section.gutter = Cm(0)           # 装订线设为0
    
    section.page_width = Cm(21.0)    # A4纸横向宽度
    section.page_height = Cm(14.85*2)  # A4纸高度的一半
    
    
    # 保存修改后的文档
    doc.save(output_path)

# 使用示例
input_file = "F:\zhou\崩铁 剧情\z_outpdf\info.docx"
output_file = "F:\zhou\崩铁 剧情\z_outpdf\info_book_fold.docx"
set_book_fold_margins(input_file, output_file)
print(f"文档已保存为 {output_file}，页边距已设置为书籍折页格式")